import kagglehub
import os
import shutil

# 获取当前脚本所在目录（scripts目录）
current_script_dir = os.path.dirname(os.path.abspath(__file__))
# 项目根目录（scripts的父目录）
project_root = os.path.dirname(current_script_dir)

# 下载数据集
dataset_path = kagglehub.dataset_download("selukzdemir/sem-nffa-europe")
print("原始下载路径:", dataset_path)

# 定义目标文件夹（项目根目录下的datasets）
target_dir = os.path.join(project_root, "datasets")

# 创建目标文件夹（如果不存在）
os.makedirs(target_dir, exist_ok=True)

# 获取数据集文件夹名称
dataset_folder = os.path.basename(dataset_path)

# 目标路径
target_path = os.path.join(target_dir, dataset_folder)

# 如果目标路径已存在，先删除
if os.path.exists(target_path):
    shutil.rmtree(target_path)
    print(f"已删除已存在的 {target_path}")

# 移动文件夹
shutil.move(dataset_path, target_dir)
print(f"数据集已移动到: {os.path.join(target_dir, dataset_folder)}")
